-
Notifications
You must be signed in to change notification settings - Fork 3.4k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[QNN] Implement quantized avg_pool2d #15057
Conversation
kparzysz-quic
commented
Jun 7, 2023
- qnn.avg_pool2d is integrated into Relay and integration tests are added
- FQ2I is modified to pick up qnn.avg_pool2d
- Canonicalization and layout conversion for qnn.avg_pool2d are implemented
- Dynamic PoolArea computation is implemented for both qnn and slice_op avg_pool2d to exclude padding area in averaging calculation when count_include_pad=False
Thanks for contributing to TVM! Please refer to the contributing guidelines https://tvm.apache.org/docs/contribute/ for useful information and tips. Please request code reviews from Reviewers by @-ing them in a comment.
Generated by tvm-bot |
3 similar comments
Thanks for contributing to TVM! Please refer to the contributing guidelines https://tvm.apache.org/docs/contribute/ for useful information and tips. Please request code reviews from Reviewers by @-ing them in a comment.
Generated by tvm-bot |
Thanks for contributing to TVM! Please refer to the contributing guidelines https://tvm.apache.org/docs/contribute/ for useful information and tips. Please request code reviews from Reviewers by @-ing them in a comment.
Generated by tvm-bot |
Thanks for contributing to TVM! Please refer to the contributing guidelines https://tvm.apache.org/docs/contribute/ for useful information and tips. Please request code reviews from Reviewers by @-ing them in a comment.
Generated by tvm-bot |
9495547
to
894d3ab
Compare
Thanks Masa! I'm planning to address your comments this weekend. |
* qnn.avg_pool2d is integrated into Relay and integration tests are added * FQ2I is modified to pick up qnn.avg_pool2d * Canonicalization and layout conversion for qnn.avg_pool2d are implemented * Dynamic PoolArea computation is implemented for both qnn and slice_op avg_pool2d to exclude padding area in averaging calculation when count_include_pad=False Co-authored-by: Fateme Hosseini <quic_fhossein@quicinc.com> Co-authored-by: Jyotsna Verma <jverma@quicinc.com> Co-authored-by: Anirudh Sundar <quic_sanirudh@quicinc.com> Co-authored-by: Venkat Rasagna Reddy Komatireddy <quic_rasagna@quicinc.com>
Merge test_wo_qnn_canonicalization.py into test_qnn_op_integration.py.
dd5176d
to
a92bbcf
Compare
Rebased to fix the GPU build error. Edit: also folded the linter fixes into the previous commit. |